torchrun --nproc_per_node=2 --nnodes=2 --node_rank=0 --master_addr=dist-example-0.dist-example.default.svc --master_port=1234 /workspace/test-dist.py
